package com.tensquare.base.service;

import com.tensquare.base.dao.LabelDao;
import com.tensquare.base.pojo.Label;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import util.IdWorker;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;
import java.util.ArrayList;
import java.util.List;

/**
 * Created by 队长 on 2018/11/25.
 */
@Service
@Transactional
public class LabelService {
    @Autowired
    private LabelDao labelDao;
    @Autowired
    private IdWorker idWorker;


    public List<Label> findAll(){
        return labelDao.findAll();
    }

    public Label findById(String id){
        return labelDao.findById(id).get();
    }

    public void save(Label label){
        label.setId(idWorker.nextId()+"");
        labelDao.save(label);
    }

    public void update(Label label){
        labelDao.save(label);
    }

    public void delete(String id){
        labelDao.deleteById(id);
    }

    public List<Label> findSearch(Label label) {
        return labelDao.findAll(new Specification<Label>() {
            /**
             * @param root 根对象 就是要把条件封装到那个对象。where 类名=label里面的参数
             * @param criteriaQuery 封装的都是查询条件 比如 group by  order by
             * @param criteriaBuilder 用来封装对象的 如果没有条件返回null 表示不需要任何条件
             * @return
             */
            @Override
            public Predicate toPredicate(Root<Label> root, CriteriaQuery<?> criteriaQuery, CriteriaBuilder criteriaBuilder) {
                // new一个list集合 来存放所有的条件
                List<Predicate> labelList = new ArrayList<>();
                if (label.getLabelname() != null && !"".equals(label.getLabelname())) {
                    Predicate predicate = criteriaBuilder.like(root.get("labelname").as(String.class), "%" + label.getLabelname() + "%");
                    labelList.add(predicate);
                }
                if (label.getState() != null  && !"".equals(label.getState())){
                    Predicate predicate = criteriaBuilder.equal(root.get("state").as(String.class), label.getState());
                    labelList.add(predicate);
                }
                // new一个数组封装数据作为最终返回的条件
                Predicate[] predicates = new Predicate[labelList.size()];
                // 把list封装成数据
                predicates = labelList.toArray(predicates);
                return criteriaBuilder.and(predicates);
            }
        });
    }


    public Page<Label> pageQuery(Label label, int page, int size) {
        // 封装一个分页对象
        Pageable pageable = PageRequest.of(page-1,size);
        return labelDao.findAll(new Specification<Label>() {
            /**
             * @param root 根对象 就是要把条件封装到那个对象。where 类名=label里面的参数
             * @param criteriaQuery 封装的都是查询条件 比如 group by  order by
             * @param criteriaBuilder 用来封装对象的 如果没有条件返回null 表示不需要任何条件
             * @return
             */
            @Override
            public Predicate toPredicate(Root<Label> root, CriteriaQuery<?> criteriaQuery, CriteriaBuilder criteriaBuilder) {
                // new一个list集合 来存放所有的条件
                List<Predicate> labelList = new ArrayList<>();
                if (label.getLabelname() != null && !"".equals(label.getLabelname())) {
                    Predicate predicate = criteriaBuilder.like(root.get("labelname").as(String.class), "%" + label.getLabelname() + "%");
                    labelList.add(predicate);
                }
                if (label.getState() != null  && !"".equals(label.getState())){
                    Predicate predicate = criteriaBuilder.equal(root.get("state").as(String.class), label.getState());
                    labelList.add(predicate);
                }
                // new一个数组封装数据作为最终返回的条件
                Predicate[] predicates = new Predicate[labelList.size()];
                // 把list封装成数据
                predicates = labelList.toArray(predicates);
                return criteriaBuilder.and(predicates);
            }
        },pageable);
    }
}
